package main

import (
	"container/heap"
	"fmt"
)

/*
go中实现优先级队列有三个步骤
1. 定义一个切片的别名（例如IntHeap）
2. 这个IntHeap实现sort.Interface接口的三个方法
3. 这个IntHeap实现heap.Interface接口的两个方法
*/

type IntHeap []int

//覆写sort.Interface接口的排序方法（规定priQueue按照什么规则排序）

func (h IntHeap) Len() int           { return len(h) }
func (h IntHeap) Less(i, j int) bool { return h[i] < h[j] }
func (h IntHeap) Swap(i, j int)      { h[i], h[j] = h[j], h[i] }

//覆写heap.Interface接口的入队出队方法

//Push 入队列
func (h *IntHeap) Push(x interface{}) {
	//传入的h是一个地址，使用*h获取到原本的数组
	*h = append(*h, x.(int))
}

//Pop 出队列
func (h *IntHeap) Pop() interface{} {
	n := len(*h)
	x := (*h)[n-1]  //首先拿到尾部的元素
	*h = (*h)[:n-1] //去除尾部的元素
	return x
}

func main() {
	h := &IntHeap{2, 1, 5, 15, 3, 6, 4, 5}
	heap.Init(h) //初始化堆
	heap.Push(h, 3)
	fmt.Printf("minimum: %d\n", (*h)[0])

	heap.Push(h, 17)

	for h.Len() > 0 {
		fmt.Printf("%d ", heap.Pop(h))
	}
	/*
		minimum: 1
		1 2 3 3 4 5 5 6 15 17
	*/
}
